tensor([[0.1000, 0.4300, 0.0000, 0.9900, 0.6900],
        [0.9000, 0.5700, 1.0000, 0.0100, 0.3100]], grad_fn=<RoundBackward1>)